import os

# 获取用户输入
old_substring = input("请输入要替换的旧字符串（例如 'av'）：")
new_substring = input("请输入新的字符串（例如 'dp'）：")

# 遍历目录及子目录
for root, dirs, files in os.walk('.'):
    for file in files:
        # 检查文件扩展名
        if file.endswith(('.c', '.h', '.r')):
            new_file_name = file

            # 替换文件名中的旧字符串为新字符串
            new_file_name = new_file_name.replace(old_substring, new_substring)

            # 如果文件名发生变化，进行重命名
            if new_file_name != file:
                old_file_path = os.path.join(root, file)
                new_file_path = os.path.join(root, new_file_name)

                # 执行重命名
                os.rename(old_file_path, new_file_path)
                print(f"Renamed: {old_file_path} -> {new_file_path}")

# 在程序执行完后，等待用户按回车键再关闭窗口
input("所有文件已重命名。按回车键退出...")
